Add code to compute the correlation matrix