diff --git a/main.cpp b/main.cpp
index 77af16b..8c3837f 100644
--- a/main.cpp
+++ b/main.cpp
@@ -3,17 +3,15 @@
 using namespace microgradCpp;
 int main()
 {
-    // DatasetType dataset = get_iris();
+
     DataFrame df;
     df.from_csv("./data/iris.csv");
-   df.normalize();
+    df.normalize();
     df.encode_column("variety");
     df.print();
     df.shuffle();
     df.print();
-    // stop();
-    // return 0;
-    // shuffle(dataset);
+
     double TRAIN_SIZE{0.8};
     // Create MLP model
     // Input: 4 features, hidden layers: [7,7], output: 3 classes