Updated visualization module comments