Added comments to the classes