Load in visualization code