Updated the test cases