Update docs based on our refactoring.