adding probability variable not included

adding probability variable not included