New feature request: output the probability array together with random walk

[dli12]

Could it also generate the probability 2D array, together with the 2D random walk array?

Thanks