Improve `GraphBuilder`
remove redundant module