Here are my changes to the escape probability part
Trying to make a change