plot_correlation function

Plot correlation