Seperated the random walk algorithm to it's own class. Like with the probability calculating random walk before.