Add visitor to the AST
Fix short mistake