Updated the visitor class

If node is done (use as example!)