Add functionality to visitor