documentation, small refactor