adding return_corr