Update docs to reflect refactoring