Add a little state machine logic.