Updated ch01.py according to instructions in Homework 1 assignment