added script to generate heatmaps of distributions