Updates comments to match docs