diff --git a/dataset-stats.py b/dataset-stats.py
index 41cfe57..5a3510f 100644
--- a/dataset-stats.py
+++ b/dataset-stats.py
@@ -24,7 +24,7 @@ def write_to_file(file_name):
         f.write("\n\n")
 
         f.write("Size of the dataset\n")
-        f.write(str(df.info()))
+        print(df.info(), file=f)
 
         f.write("\n\n")
 
@@ -39,7 +39,7 @@ def write_to_file(file_name):
         f.write("\n\n")
 
         f.write("Size of undersampled dataset\n")
-        f.write(str(undersample_data.info()))
+        print(undersample_data.info(), file=f)
 
         f.write("\n\n")
 
@@ -56,7 +56,7 @@ def write_to_file(file_name):
         f.write("\n\n")
 
         f.write("Statistical measures of the training dataset of whole data\n")
-        f.write(str(pd.concat([X_train, y_train], axis=1).info()))
+        print(pd.concat([X_train, y_train], axis=1).info(), file=f)
         f.write("\n")
         f.write(str(pd.concat([X_train, y_train], axis=1).describe()))
         f.write("\n")
@@ -65,7 +65,7 @@ def write_to_file(file_name):
         f.write("\n\n")
 
         f.write("Statistical measures of the test dataset of whole data\n")
-        f.write(str(pd.concat([X_test, y_test], axis=1).info()))
+        print(pd.concat([X_test, y_test], axis=1).info(), file=f)
         f.write("\n")
         f.write(str(pd.concat([X_test, y_test], axis=1).describe()))
         f.write("\n")
@@ -74,8 +74,8 @@ def write_to_file(file_name):
         f.write("\n\n")
 
         f.write("Statistical measures of the training dataset of undersampled data\n")
-        f.write(
-            str(pd.concat([X_train_undersample, y_train_undersample], axis=1).info())
+        print(
+            pd.concat([X_train_undersample, y_train_undersample], axis=1).info(), file=f
         )
         f.write("\n")
         f.write(
@@ -95,7 +95,9 @@ def write_to_file(file_name):
         f.write("\n\n")
 
         f.write("Statistical measures of the test dataset of undersampled data\n")
-        f.write(str(pd.concat([X_test_undersample, y_test_undersample], axis=1).info()))
+        print(
+            pd.concat([X_test_undersample, y_test_undersample], axis=1).info(), file=f
+        )
         f.write("\n")
         f.write(
             str(pd.concat([X_test_undersample, y_test_undersample], axis=1).describe())