Added correlation part