Changed probability calculation