Add Optree AST class