Added a map to the AST node

So I don't destroy the AST JSON when it's time to write it back