added test
code refactoring