updated visualizer to deal with imports in a sane way