Move timestep parameter
Simplify example