Added comments to the class