Update docs after refactoring