Updated the test cases to pass.