small refactoring style code