Create hw6.py their answers