Refactor Visualizer class and add some comments