Change heatmap so that it's size scales with input data. Also shrink default cell size