Add heatmap code