Various cleanups and documentation improvements for SmallSet and Prob