updated code for column-count as there is new CSS