diff --git a/csaxs_bec/bec_widgets/widgets/client.py b/csaxs_bec/bec_widgets/widgets/client.py
index f4a9fbb..ec2d37c 100644
--- a/csaxs_bec/bec_widgets/widgets/client.py
+++ b/csaxs_bec/bec_widgets/widgets/client.py
@@ -87,14 +87,14 @@ class XRayEye(RPCBase):
     @rpc_call
     def enable_live_view(self):
         """
-        None
+        Get or set the live view enabled state.
         """
 
     @enable_live_view.setter
     @rpc_call
     def enable_live_view(self):
         """
-        None
+        Get or set the live view enabled state.
         """
 
     @property
@@ -138,3 +138,11 @@ class XRayEye(RPCBase):
         """
         None
         """
+
+
+class XRayEye2DControl(RPCBase):
+    @rpc_call
+    def remove(self):
+        """
+        Cleanup the BECConnector
+        """
diff --git a/csaxs_bec/bec_widgets/widgets/xray_eye/x_ray_eye_plugin.py b/csaxs_bec/bec_widgets/widgets/xray_eye/x_ray_eye_plugin.py
index fb94732..f611e47 100644
--- a/csaxs_bec/bec_widgets/widgets/xray_eye/x_ray_eye_plugin.py
+++ b/csaxs_bec/bec_widgets/widgets/xray_eye/x_ray_eye_plugin.py
@@ -1,10 +1,10 @@
 # Copyright (C) 2022 The Qt Company Ltd.
 # SPDX-License-Identifier: LicenseRef-Qt-Commercial OR BSD-3-Clause
 
+from bec_widgets.utils.bec_designer import designer_material_icon
 from qtpy.QtDesigner import QDesignerCustomWidgetInterface
 from qtpy.QtWidgets import QWidget
 
-from bec_widgets.utils.bec_designer import designer_material_icon
 from csaxs_bec.bec_widgets.widgets.xray_eye.x_ray_eye import XRayEye
 
 DOM_XML = """