Update generater.py