diff --git a/Objet/__pycache__/display.cpython-311.pyc b/Objet/__pycache__/display.cpython-311.pyc
index e34941d41697f6c341f6bf9d56f46a8546b36e41..e338aeb771cf6f31bc02f04a2a5d165bef4efae2 100644
Binary files a/Objet/__pycache__/display.cpython-311.pyc and b/Objet/__pycache__/display.cpython-311.pyc differ
diff --git a/Objet/display.py b/Objet/display.py
index 60f08b51a7e2833d5d21eb794b382b7c6e1d32ae..734479a713556417b263f6f96305940074c22cdb 100644
--- a/Objet/display.py
+++ b/Objet/display.py
@@ -92,7 +92,10 @@ class Display():
         self.draw_line(point3, point4, color)
         self.draw_line(point4, point1, color)
 
-    def display(self):
+    def display(self, brightness):
+        self.draw_shape(brightness)
+        self.draw_all_points(brightness)     
+        self.draw_traj()  
         cv2.imshow('main', self.frame)
     
     
diff --git a/Objet/main.py b/Objet/main.py
index 108f4c8f2f17b569a3cd858e194e0bfce1d47d0d..f4e687c1ddda29fc0b0b828878d0358031cc002e 100644
--- a/Objet/main.py
+++ b/Objet/main.py
@@ -60,12 +60,8 @@ class Main:
                 darken_factor = 0.5 
                 self.display.frame = (self.display.frame * darken_factor).astype(np.uint8)
 
-
-            self.display.draw_shape(self.ui.brightness)
-            self.display.draw_all_points(self.ui.brightness)
-            
-            self.display.draw_traj()   
-            self.display.display()
+ 
+            self.display.display(self.ui.brightness)
             #self.session.load(self.display.frame)
             #cv2.imshow('main', im_proc.processed_frame_difference)
 
diff --git a/success_probability_vs_E.png b/success_probability_vs_E.png
index d94ab99bf4f7a157fa8211b664ad5912e5b628f2..79bd413b21f4d89c0ceaab57923b5ac49b0d7101 100644
Binary files a/success_probability_vs_E.png and b/success_probability_vs_E.png differ