Added update + Unit test for update
Added update + Unit test for update