3D visualization bugfixes