Updated visualization method