Added heatmap png and script