Update task_01.py

got idea from Quiz 6