Fixed bug that allows for >1 probabilities