Added some initial AST support