Simple jquery function to equalise the height of elements