UI polish and 3D visualization