Update ch01.py