Update hw8sol.py