Add some tests for Update