Added some comments into the classes