Update firstgen_example.py