Fixed ex1.py