Added code subgraph extraction