Annotate parsed AST with locations