Update gen_example.py

Reverted back to original because I'm an idiot.