Update Generate.py