Update hw0.py