update ex5.py