write all tests for update, fixes for remove