Add comments to class