script for heatmap