Update gen_grammar.py