diff --git a/src/projet8.py b/src/projet8.py
index c4aac1b..9205a92 100644
--- a/src/projet8.py
+++ b/src/projet8.py
@@ -9,7 +9,7 @@ from pyclustering.cluster.kmedoids import kmedoids
 from pyclustering.cluster import cluster_visualizer
 from pyclustering.utils import read_sample
 from pyclustering.samples.definitions import FCPS_SAMPLES
-
+import matplotlib.pyplot as plt
 
 class Conformations:
     """
@@ -156,31 +156,34 @@ class Conformations:
         x = np.arange(nb_confs)
         y = np.zeros(nb_confs)
 
-        for i in range(len(clusters_diss)):
-            for j in clusters_diss[i]:
+        for i in range(len(clusters)):
+            for j in clusters[i]:
                 y[j] = i+1
         
         color = []
+        #dict_col = {1:'b',2:'c',3:'',4:'',5:'',6:''}
         for i in y:
             if i == 1:
                 color.append('b')
             if i == 2:
-                color.append('c')
-            if i == 3:
                 color.append('g')
-            if i == 4:
+            if i == 3:
                 color.append('y')
-            if i == 5:
+            if i == 4:
                 color.append('r')
+            if i == 5:
+                color.append('c')
             if i == 6:
                 color.append('m')
-            else:
+            elif i > 6:
                 color.append('k')
 
         plt.bar(x, y, width=1.0, color=color)
         
         if output_name != None:
-            plt.savefig(output_name)
+            plt.savefig("results/"+output_name)
+
+        plt.close()
 
 if __name__ == "__main__":
     if len(sys.argv) < 2: