Update test cases

Update the test cases to be useful!