Adding update hook to revert feature