small changes to the heatmap