Change signature for the update method