modified the heatmap