add unit test logic