Added some comments for the classes