Some improvements, hooked up timestep to be a parameter.