add graph plotting to utils