Fix comments to match functions