code: function(count) {
      columnHeights = [];
      for(var i = 0; i < count; i++) {
        columnHeights.push(0);
      }
      cellsContainer.style.width = (count * (COLUMN_WIDTH + GAP_WIDTH) - GAP_WIDTH) + 'px';
    }
docs: Reset array of column heights and container width.