Create simulation.py