add unit test code