Equal heights JS tweak