Updated visualization editor code.