update on hw6.py
update to hw6