Add correlation option to Heatmap