seemingly getting prob dists correctly