Add `loc` to AST nodes