diff --git a/dane.py b/dane.py
index 34f43e3..1283c93 100644
--- a/dane.py
+++ b/dane.py
@@ -140,3 +140,9 @@ test["HighRiskLastYear"].isna().sum()
 test.info()
 train.info()
 valid.info()
+
+cat_columns = test.select_dtypes(['category']).columns
+
+test.to_csv("test.csv")
+train.to_csv("train.csv")
+valid.to_csv("valid.csv")
\ No newline at end of file