The summation of the probabiiities for p$\_$state is noI $1$ as expected due to only showing $6$ probabilities not ten or $15.$ Please advise how can I modify the code to make the sum of p$\_$state probabilities one? Thank you.