Update documentation to follow recent refactoring