Probabilites are not being calculated correctly