Update HW1_class.py

revised after get a comments from Dr.Hetland