Update to the Viz class