Update ex4.py