Updated visualization subsystem. Visualizations now work end-to-end