Still need to add test for visualizer