Added test for update