Added a test for update