Update simulate.py