Add script to create heatmaps