Add new changes to random walk notebook