Almost done with dfa.py