Update correlation2.py