Optimised Simulation.py