diff --git a/randomfiletree/cli.py b/randomfiletree/cli.py
index 0d3247c..185857e 100644
--- a/randomfiletree/cli.py
+++ b/randomfiletree/cli.py
@@ -16,26 +16,30 @@ def parser():
         "--directory-probability",
         default=1,
         dest="prob_folder",
-        help="Probability to create a folder"
+        help="Probability to create a folder",
+        type=float
     )
     parser.add_argument(
         "-f",
         "--file-probability",
         default=1,
         dest="prob_file",
-        help="Probability to create a file"
+        help="Probability to create a file",
+        type=float
     )
     parser.add_argument(
         "-r",
         "--repeat",
-        default=10,
+        default=2,
         help="Number of times to traverse existing file/directory structure to "
-             "create new elements"
+             "create new elements",
+        type=int
     )
     parser.add_argument(
         "--maxdepth",
         default=None,
-        help="Maximal depth of file/directory structure to create"
+        help="Maximal depth of file/directory structure to create",
+        type=int
     )
     return parser
 
diff --git a/randomfiletree/core.py b/randomfiletree/core.py
index bb6e15a..a9299b0 100644
--- a/randomfiletree/core.py
+++ b/randomfiletree/core.py
@@ -45,6 +45,7 @@ def create_random_tree(basedir, prob_file=2, prob_folder=1, repeat=1,
     allfiles = []
     for i in range(repeat):
         for root, dirs, files in os.walk(str(basedir)):
+            print(root)
             if random.random() < prob_folder:
                 p = Path(root) / random_string()
                 p.mkdir(exist_ok=True)
diff --git a/randomfiletree/test/test_cli.py b/randomfiletree/test/test_cli.py
index 1858de8..0ebbbd5 100644
--- a/randomfiletree/test/test_cli.py
+++ b/randomfiletree/test/test_cli.py
@@ -20,6 +20,6 @@ def test_cli_module(self):
     def test_parser(self):
         p = parser()
         with tempfile.TemporaryDirectory() as dirname:
-            cli(p.parse_args(dirname))
-            cli(p.parse_args(dirname, "-f", 0.5, "-d", 3, "-r", 10))
-            cli(p.parse_args(dirname, "-f", 0.5, "-d", 3, "--maxdepth", 5))
+            cli(p.parse_args([dirname]))
+            cli(p.parse_args([dirname, "-f", "0.5", "-d", "3", "-r", "3"]))
+            cli(p.parse_args([dirname, "-f", "0.5", "-d", "3", "--maxdepth", "2"]))