forgot documentation, small refactor