Added correlation plot