Updates for visualization submodule.