Update simulation_example.py