cleaning up Visualize