Add an updating function