diff --git a/src/main/java/edu/ntnu/stud/chaosgame/controller/game/GuiButtonController.java b/src/main/java/edu/ntnu/stud/chaosgame/controller/game/GuiButtonController.java
index 2fec05e53af79cd512a9a7ae1dcbe5e2b49106ed..74c66aeefd3bf38cf10745e4511e65b48123f9dd 100644
--- a/src/main/java/edu/ntnu/stud/chaosgame/controller/game/GuiButtonController.java
+++ b/src/main/java/edu/ntnu/stud/chaosgame/controller/game/GuiButtonController.java
@@ -74,8 +74,12 @@ public class GuiButtonController {
         gui.getNewButton().setOnAction(event -> newGame());
         gui.getClearButton().setOnAction(event -> clearImageView());
         gui.getQuitButton().setOnAction(event -> quitGame());
+        gui.getColorCheckBox().setOnAction(event -> game.setUseColor(gui.getColorCheckBox().isSelected()));
+
     }
 
+
+
     /**
      * Update the description of the chaos game.
      * TODO: this method may need to be changed depending on how we implement the UI. Rename to update?
diff --git a/src/main/java/edu/ntnu/stud/chaosgame/view/ChaosGameGui.java b/src/main/java/edu/ntnu/stud/chaosgame/view/ChaosGameGui.java
index b773bcbc0a523573a03db045b6e378cc592e10b7..5feb6a4f92cd9cfdedd9ef7e499d15ae85422345 100644
--- a/src/main/java/edu/ntnu/stud/chaosgame/view/ChaosGameGui.java
+++ b/src/main/java/edu/ntnu/stud/chaosgame/view/ChaosGameGui.java
@@ -332,7 +332,7 @@ public ChaosGameGui(Stage primaryStage) throws IOException {
         improvedBarnsleyButton);
 
     sideMenu.getChildren().addAll(separator1, colorBox, separator2);
-    this.initializeColorButtonHandler();
+    //this.initializeColorButtonHandler();
 
     // Add parameter VBox
     sideMenu.getChildren().add(parameterBox);