Edited correlate.py