Simulation pasts initial tests
Added clean data function, tests pass