added code for plot of correlation matrix