Update chapter_1.py