Added add_probability!
Merged