HW2: add the code file hw2.py to the submited folder