Update homework1.4/task.py