fixing heatmap layout