Added a lot boiler code to make the viz happen, though I don't
necessarily know what those lines of code are doing