Add more correlation code