Fix a bug in random_step.py
minor updates on docs