Add visualization-stuff