Update HW1_class.py

Added some comments inside the class to make the class more understandable.