small refactor
edit version