correction in plot_corr call at the end