minor changes heatmap