Working on Probability using Pandas. Need to confirm that the code is correct.