Added tests for the graph