Update task2.py