added the plot_timesteps function, changed a few things for simulation parameters