Correction of task_04.py