Add Revert functionality