Commented the minor changes I made to the class.