moved visualization code to its own folder