Updated with random_walk