Update random_walk.py