Added tests of plotting. Conflicted about it though because it adds so much time to the tests, for a pretty marginal gain. Would be nice to separate the plotting functionality so that I don't actually have to test that function.