Refactor. 2 Unit Test sample