update ex2.py