Added ability to run Prob with only the document distribution.