add the re-written sampler