Added some comments to the classes