Add additional callback for AST construction support