Minor cleanup for setElementsHeight - using .css instead of .map which made no sense