Update Ch6_ex2.py