Add initial support for item count probability distributions