Add heatmap plot