Update docs to match updated interface