added heatmap script