Fighting with AST code generation
Working out on the AST