Adding heatmap