Adding an automaton should not involve changing the main. Rather, there should be a config file or even just a .py file which is imported by the main. Much better. The default automaton should also be defined there.