modify ex1.py