minor changes to the Visualize class