a little change to ch08/04.py