small changes in class Visualize