would it be feasible to move the walk mode into build_graph