diff --git a/DisplayWindow.py b/DisplayWindow.py
index c4f363662b841a0b2fcda253b6c5d6672beba513..3c251f64b44d94f194cc5beb5dc68685e324b4a8 100644
--- a/DisplayWindow.py
+++ b/DisplayWindow.py
@@ -106,7 +106,7 @@ class Advanced(QMainWindow):
         self.check_labels = ['Cut', 'Colour', 'Clarity', 'Carat', 'Depth', 'Table', 'X', 'Y', 'Z']
         self.graph_selector_X.addItems((self.check_labels + ['price']))
         self.graph_selector_Y.addItems((self.check_labels + ['price']))
-        self.regression_model.addItems(["Linear Regression", "Gradient Descent", "Random Forest"])
+        self.regression_model.addItems(["Linear Regression", "XG Boost", "XGBRegressor", "Neural Network"])
         self.checkboxes = []
 
 
@@ -127,8 +127,11 @@ class Advanced(QMainWindow):
             self.checkboxes.append(checkbox)  # Store checkboxes in the list
 
     def handle_checkbox_state(self):
+        RegressionModel.Advanced_selections = []
         for i, checkbox in enumerate(self.checkboxes):
             state = checkbox.checkState()
+            if state == Qt.CheckState.Unchecked:
+                RegressionModel.Advanced_selections.append(checkbox.text())
             RegressionModel.Advanced_selections[checkbox.text()] = (state == Qt.CheckState.Checked)