Added probability support to collections