Completed q6git add hw1.py