Update ex1.py