add heatmap plot