Update visualize.m