changed code in hw.py