add small refactoring
refactoring
refactoring