Update docs to reflect our refactoring