Update cw07-06.py