Fixing chap01soln.py