Update ex3.py