Minor edit visitor

Enhancing the code snippet.