I removed this yesterday. Don't plot outside of test_viz.py