exercise_2.py