Fixed issues with probability scoring