Hmm, this was actually a very intentional design decision on my part, because the plotting code is very specific to the data produced by the sampler (and if the external function takes the sampler object as the first argument, it doesn't seem to me to be that much different than having it be a method, where the first argument is `self`, which is the sampler). Could you explain a bit more why moving it outside of the class would be better?