Got all AST tests working