"FinalProbability class changes and test"