adding the heatmap code