Use a DiGraph