fixed formatting in task_01.py