Fixed issue with probability calculation and completed test case