Add unit test for update