Updated the input test case