Added code to track probability of simulation