Add PKB callers for new functions in AST