# Define nodes
builder = StateGraph(State)
builder.add_node(a)
builder.add_node(b)