Add refreshCellHeights method