Update chapter_9.py