Minor fixes for visualizer

@rampantmonkey, colud you check everything looks ok?