Update sim.py