adding more tests to AST