Update state transitions to refer to mode of operation instead of input value