simulation.py should now be working code