Update simulation.jl