Update lesson 1 task 1.py

adjusted for runability