changes to heatmap figure