simulation.py updated and should work