The other function now works for the update method