help: improve Env.step example. This fixes #1143