Update ch2.py