Added some tests to check multiple states of probability distributions
Merged