Add update hook