Added lots of visualizer code.
UI looks better now