removing unneeded argument from the visualize() function