updated plot_random_walks function