add visualization flag and visual_example