Add support for both variants of geometric distribution