Add script to visualize AST