EXERCISE: Python - Correlation