Now with probability calculation