Added heatmap code