Add a test case to verify update