Better Demo Example

[EthanJamesLew]

• set a seed so the example reproduces
• train on more trajectories for more stable output

[stanleybak]

also with the example for me the progress bar gets to 50% when the problem exits