added some plotting to the graph