Update HW1_function.py