updated viz class