Updated height of cells on small screens