update ex1.py