hw6.py their answers