hw01 finished ch01.py changes