Updated UI layer to work regardless if the camera is 2d or 3d