minor changes to heatmap.