Update to hw02.py