update task4.py