Update docs to match new API