Update simulate.py

Whoops. Forgot to update that part.