Working grammar but broke AST