Re-add heatmap