Added the probability calculation

Set as a separate method, might want to add formatting in the next revision! :D