Update make_heatmaps.R