completed the probability section of the assignment