Hello @pranalipatel,
Sorry, I spoke too soon again! On my website, any elements with 'equal height of items in wrap' are all displayed incorrectly as a result of the above code. I believe this is a theme issue with BeTheme, or with the Muffin Builder that comes with the theme. It is a pity because the code works perfectly otherwise. For now I have not used the code until a fix for this issue is found.
Thanks for helping me out though!