added comments to the classes