fix visualize.R