added heatmap plot