added viz code