Added correlation heatmap.