Heatmap rendering using matplotlib