Added unit test cases.