Modify on DFA.py