Added unit test cases