diff --git a/__main__.py b/__main__.py
index e69de29..6e564d9 100644
--- a/__main__.py
+++ b/__main__.py
@@ -0,0 +1,5 @@
+from src.logic import userInterface
+
+
+if __name__ == "__main__":
+    userInterface.launch_gui()
diff --git a/main.py b/main.py
index 5977a81..7f6aa25 100644
--- a/main.py
+++ b/main.py
@@ -1,20 +1,10 @@
 from src.logic import userInterface
-from src.ui.widgets.graph import GraphWidget
 
 
-def display_graph():
-    import sys
-    from PyQt6 import QtWidgets
-    app = QtWidgets.QApplication(sys.argv)
-    data = {
-        "x": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
-        "y": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
-        "y2": [10, 9, 8, 7, 6, 5, 4, 3, 2, 1],
-    }
-    widget = GraphWidget(data=data, legend_labels=["+", "-"])
-    widget.show()
-    sys.exit(app.exec())
 
 if __name__ == '__main__':
     userInterface.launch_gui()
+
+    
+    
     # display_graph()
\ No newline at end of file