Update hw.py

My changes