Clean up some of the probability material