Update documentation comments to match current implementation