fix docs
refactoring