Add reminder to refactor
Move probability code into root route