updated viz code