minor changes to class Visualize