Added a button to move one step.
Switched the goal state so that it starts a new episode immediately.