Unit testing
Added update + Unit test for update