You wouldn't extend this module, because it does what it's supposed to. What you're describing would be achieved by getting the graph object back, then mutating it with further inputs from somewhere, like the regex debugger mentioned elsewhere here.