Add visitor for ast visitng