Add PKB APIs for omitted functions in AST