diff --git a/sample/main.cc b/sample/main.cc
index ee453ca..de0e928 100644
--- a/sample/main.cc
+++ b/sample/main.cc
@@ -259,30 +259,30 @@ int main(int argc, char** argv)
     }
     states[className] = vector<int>(
         maxes[className]);
-    cout << "****************** KDB ******************" << endl;
-    auto kdb = bayesnet::KDB(2);
-    kdb.fit(Xd, y, features, className, states);
-    for (auto line : kdb.show()) {
-        cout << line << endl;
-    }
-    cout << "Score: " << kdb.score(Xd, y) << endl;
-    cout << "****************** KDB ******************" << endl;
-    cout << "****************** SPODE ******************" << endl;
-    auto spode = bayesnet::SPODE(2);
-    spode.fit(Xd, y, features, className, states);
-    for (auto line : spode.show()) {
-        cout << line << endl;
-    }
-    cout << "Score: " << spode.score(Xd, y) << endl;
-    cout << "****************** SPODE ******************" << endl;
-    cout << "****************** AODE ******************" << endl;
-    auto aode = bayesnet::AODE();
-    aode.fit(Xd, y, features, className, states);
-    for (auto line : aode.show()) {
-        cout << line << endl;
-    }
-    cout << "Score: " << aode.score(Xd, y) << endl;
-    cout << "****************** AODE ******************" << endl;
+    // cout << "****************** KDB ******************" << endl;
+    // auto kdb = bayesnet::KDB(2);
+    // kdb.fit(Xd, y, features, className, states);
+    // for (auto line : kdb.show()) {
+    //     cout << line << endl;
+    // }
+    // cout << "Score: " << kdb.score(Xd, y) << endl;
+    // cout << "****************** KDB ******************" << endl;
+    // cout << "****************** SPODE ******************" << endl;
+    // auto spode = bayesnet::SPODE(2);
+    // spode.fit(Xd, y, features, className, states);
+    // for (auto line : spode.show()) {
+    //     cout << line << endl;
+    // }
+    // cout << "Score: " << spode.score(Xd, y) << endl;
+    // cout << "****************** SPODE ******************" << endl;
+    // cout << "****************** AODE ******************" << endl;
+    // auto aode = bayesnet::AODE();
+    // aode.fit(Xd, y, features, className, states);
+    // for (auto line : aode.show()) {
+    //     cout << line << endl;
+    // }
+    // cout << "Score: " << aode.score(Xd, y) << endl;
+    // cout << "****************** AODE ******************" << endl;
     cout << "****************** TAN ******************" << endl;
     auto tan = bayesnet::TAN();
     tan.fit(Xd, y, features, className, states);
diff --git a/src/Metrics.cc b/src/Metrics.cc
index 887706a..5c87e28 100644
--- a/src/Metrics.cc
+++ b/src/Metrics.cc
@@ -127,6 +127,5 @@ namespace bayesnet {
         auto result = vector<pair<int, int>>();
         auto mst = MST(features, weights, root);
         return mst.maximumSpanningTree();
-
     }
 }
\ No newline at end of file
diff --git a/src/Mst.cc b/src/Mst.cc
index ce75376..56a0558 100644
--- a/src/Mst.cc
+++ b/src/Mst.cc
@@ -108,7 +108,6 @@ namespace bayesnet {
             }
         }
         g.kruskal_algorithm();
-        //g.display_mst();
         auto mst = g.get_mst();
         return reorder(mst, root);
     }