diff --git a/src/main/java/fr/univlille/sae/classification/model/Iris.java b/src/main/java/fr/univlille/sae/classification/model/Iris.java
index abf271abb8ef1ba9bb655bab997a826b774bb18d..b91fd287cdb198a7bbd19255d89bbaaa600dd51f 100644
--- a/src/main/java/fr/univlille/sae/classification/model/Iris.java
+++ b/src/main/java/fr/univlille/sae/classification/model/Iris.java
@@ -141,23 +141,8 @@ public class Iris extends LoadableData {
     }
 
 
-    /**
-     * Renvoie la couleur associée à la variété de l'Iris.
-     * @return couleur correspondant à la variété.
-     */
-    @Override
-    public Color getColor() {
-        switch (this.variety) {
-            case "Setosa":
-                return Color.RED;
-            case "Versicolor":
-                return Color.BLUE;
-            case "Virginica":
-                return Color.GREEN;
-            default:
-                return Color.BLACK; // Couleur par défaut si la variété est inconnue
-        }
-    }
+   
+
 
     /**
      * Renvoie les noms des attributs de l'Iris.
diff --git a/src/main/java/fr/univlille/sae/classification/model/LoadableData.java b/src/main/java/fr/univlille/sae/classification/model/LoadableData.java
index e33aa56df15fe6e6a83e71adecae981d5508889c..11b71137b6bb3e9a51491dac47ab2c132f5ed4f0 100644
--- a/src/main/java/fr/univlille/sae/classification/model/LoadableData.java
+++ b/src/main/java/fr/univlille/sae/classification/model/LoadableData.java
@@ -55,11 +55,11 @@ public abstract class LoadableData {
 
     public abstract Map<String, Object> getAttributesNames();
 
-    /**
-     * Renvoie la couleur associée à l'objet.
-     * @return couleur correspondant à la classification de l'objet.
-     */
-    public abstract Color getColor();
+
+
+
+
+
 
     /**
      * Renvoie la valeur des données en fonction de l'axe spécifié.
diff --git a/src/main/java/fr/univlille/sae/classification/model/Pokemon.java b/src/main/java/fr/univlille/sae/classification/model/Pokemon.java
index f52f5b1a285687aab16c03cb5f9ff2c41298ef8d..8c8bdb0563c7e9c04aa76cbd2a8b7d3b4acace2f 100644
--- a/src/main/java/fr/univlille/sae/classification/model/Pokemon.java
+++ b/src/main/java/fr/univlille/sae/classification/model/Pokemon.java
@@ -112,54 +112,7 @@ public class Pokemon extends LoadableData {
         return attrNames;
     }
 
-    /**
-     * Renvoie la couleur associée à l'objet.
-     *
-     * @return couleur correspondant à la classification de l'objet.
-     */
-    @Override
-    public Color getColor() {
-        switch (this.type1) {
-            case "normal":
-                return Color.LIGHTGREY;
-            case "grass":
-                return Color.GREEN;
-            case "electric":
-                return Color.YELLOW;
-            case "bug":
-                return Color.GREENYELLOW;
-            case "psychic":
-                return Color.PLUM;
-            case "poison":
-                return Color.PURPLE;
-            case "steel":
-                return Color.SILVER;
-            case "dragon":
-                return Color.WHITE;
-            case "flying":
-                return Color.SKYBLUE;
-            case "water":
-                return Color.BLUE;
-            case "rock":
-                return Color.SIENNA;
-            case "fire":
-                return Color.RED;
-            case "fairy":
-                return Color.PINK;
-            case "fighting":
-                return Color.FIREBRICK;
-            case "ice":
-                return Color.DARKTURQUOISE;
-            case "ghost":
-                return Color.DARKMAGENTA;
-            case "dark":
-                return Color.GREY;
-            case "ground":
-                return Color.KHAKI;
-            default:
-                return Color.BLACK; // Couleur par défaut si la variété est inconnue
-        }
-    }
+
 
     /**
      * Renvoie la valeur des données en fonction de l'axe spécifié.
diff --git a/src/test/java/fr/univlille/sae/classification/model/IrisTest.java b/src/test/java/fr/univlille/sae/classification/model/IrisTest.java
index e0873a5f3a78475d65d874d2e7d40ab6788a9f0a..824be5f4bcf220de6165f03d342622ef2316bb37 100644
--- a/src/test/java/fr/univlille/sae/classification/model/IrisTest.java
+++ b/src/test/java/fr/univlille/sae/classification/model/IrisTest.java
@@ -40,7 +40,7 @@ class IrisTest {
 
     @Test
     void getColor() {
-        assertEquals(Color.RED, iris.getColor());
+        assertEquals(Color.RED, iris.getClassifications().get(iris.getClassification()));
     }
 
     @Test