Update simulation.py