Added graph plotting to the test.