Update documentation after refactoring