Add comments to classes