Fixing bug in tree_grammer.py syntax definition