Simulation set up

Code from class