ENH: removed pyplot as dependency. Moved plotting code to example