Update heatmap.R