Update edify_generator.py