Fix tests.
Add graph.