Update ch8_ex6.py