Added unit test code.