diff --git a/main.py b/main.py
index 471d4f8..2719af5 100644
--- a/main.py
+++ b/main.py
@@ -1,3 +1,4 @@
+import pandas as pd
 from sklearn.model_selection import train_test_split
 from sklearn.preprocessing import MinMaxScaler
 from datasets import load_dataset
@@ -20,3 +21,7 @@ val[numerical_features] = scaler.fit_transform(val[numerical_features])
 train.dropna(inplace=True)
 test.dropna(inplace=True)
 val.dropna(inplace=True)
+
+train.to_csv('liver.train.data')
+test.to_csv('liver.test.data')
+val.to_csv('liver.dev.data')