Add visitor for displaying the AST