Script for plotting heatmaps