Add heatmaps