update HW0.py