some minor improvements in visualize