Trying to get unigram distribution instead of uniform. Trying to implement ContinuationMaximumLikelihoodEstimator.