Update ch5_ex5.py