Update task4.py