Visualization almost done