added hw6.py