Update ex2-lesson3.py