[MGTECON640] PSet 4 - Fixing broken commit

I accidentally cut out the simulation loop in the last commit…