ex1.py edited