Added visualization submodule