diff --git a/split_into_files.py b/split_into_files.py
index b689e9b8d82ae20962a884a39d575e2b4e5626b1..a640a96284062f7879e1d5523cc82b676aa8cd23 100644
--- a/split_into_files.py
+++ b/split_into_files.py
@@ -1,5 +1,10 @@
+import sys
 from decentralizepy.datasets.Femnist import Femnist
 
-f = Femnist()
 
-f.file_per_user("leaf/data/femnist/data/train", "leaf/data/femnist/per_user_data/train")
+if __name__ == "__main__":
+    f = Femnist(None, None, None)
+    assert len(sys.argv) == 3
+    frm = sys.argv[1]
+    to = sys.argv[2]
+    f.file_per_user(frm, to)