Make AST Visitor take mutable references