Added 3D visualasation