add update and remove function to UoW and its tests