BUG: add plotting to setup.py