Update module06_simulation.py