Added tests for update