Finish the visualization part

I didn't start it.