Added Output Probability calculation and a few comments