hw1.py success