Adding in a benchmark similar to figure 7. still cannot reproduce just using tree sampling...