Brought in visualization code