Add basic framework for controlling probabilities