let find_simple_stmt_nodes pdg stmt =
  let idx = Pdg.get_index pdg in
  let key = Key.stmt_key stmt in
  let nodes = FctIndex.find_all idx key in
    match stmt.skind with
      | Return _ -> (* also add OutRet *)
          (try
            let ret = FctIndex.find_all idx Key.output_key in
              ret @ nodes
          with Not_found -> nodes)
      | _ -> nodes