docs
minor refactor