Added correlation plotting