Update ch6_ex7.py