completing unit test code