Add a method to revert cell state