update generte.py