Small fixes
Visualization now uses cpp-lib/geom primitives