unit test fixing, graph implementation