Added test for sampling. Blah