add import and class in python.peg