Added heatmap plot