Update chapter2.py