Adding exercise-2.1.py