Updated visualization util.