Update HW1_class.py