Align documentation with refactored API