MAINT: move some tree sampling functions out of the rao-teh sampling module and into their own tree sampling module