Add automaton logic