hw6.py