Added script for plotting heatmap.