Added two more unit test cases