Track asgnExpr and Stmt in the body
Helper to find a subtree of a tree.