Add comments to the class