Added heatmap plots using matplotlib