Don't hardcode graph size in small_graph.py.