Add AST register definitions