update ex4.py