adding ch01.py
work during lecture 4