Move matplotlib imports inside relevant functions

[joezuntz]

Matplotlib can be a pain on some systems, and isn't needed for the sampling itself.

This small change moves the imports of matplotlib inside the functions where it's needed, so that you can import the rest of the code without it.

Cheers, Joe Zuntz

[eggplantbren]

A sensible change - merged. Thanks