Update random_walks_simulation.py