Adding automatic heatmaps code