Create correlation2.py