Minor changes in heatmap