QLS AST improvements