Simulation.py gives good results