Minor fixes to task_02.py