Minor fixes
Visualization, first try