Added update hook to revert all features