updated correlation matrix code and in paper