improving heatmap colours