Cleanup hw1.py