heatmap code added