Update chapter_3.py