Fix typo didst -> dist.
Add a sampler for the categorical distribution.