mini refactor
mini refactor