Update chapter01.py