diff --git a/node2vec_spark/src/main/scala/com/navercorp/Main.scala b/node2vec_spark/src/main/scala/com/navercorp/Main.scala
index f3494e5..9088022 100644
--- a/node2vec_spark/src/main/scala/com/navercorp/Main.scala
+++ b/node2vec_spark/src/main/scala/com/navercorp/Main.scala
@@ -39,6 +39,12 @@ object Main {
     opt[Int]("numWalks")
             .text(s"numWalks: ${defaultParams.numWalks}")
             .action((x, c) => c.copy(numWalks = x))
+    opt[Int]("dim")
+            .text(s"dim: ${defaultParams.dim}")
+            .action((x, c) => c.copy(dim = x))
+    opt[Int]("iter")
+            .text(s"iter: ${defaultParams.iter}")
+            .action((x, c) => c.copy(iter = x))
     opt[Double]("p")
             .text(s"return parameter p: ${defaultParams.p}")
             .action((x, c) => c.copy(p = x))