added the heatmap