Updates to gen_python.py