update simulation.py