Revised heatmap png and script