ENH: return graph