Several visualize fixes