Added correlation matrix and heat map