Probability Problems In Python