Add TODO to tidy up AST