Add testing for update and remove operations.