Create task 2.py