diff --git a/README.md b/README.md
index d1e8195..493408e 100644
--- a/README.md
+++ b/README.md
@@ -34,4 +34,4 @@ How to run spark program through Intellij?
 - Set main class as 
     org.apache.spark.deploy.SparkSubmit
 - Set program arguments as
-   --master local -- class <main_class> target/scala-2.12/<jar_name>
+   --master local --class <main_class> target/scala-2.12/<jar_name>
diff --git a/src/main/scala/thoughtworks/Diamond.scala b/src/main/scala/thoughtworks/Diamond.scala
index c703876..c1c3ac8 100644
--- a/src/main/scala/thoughtworks/Diamond.scala
+++ b/src/main/scala/thoughtworks/Diamond.scala
@@ -5,7 +5,11 @@ import org.apache.spark.sql.SparkSession
 
 object Diamond {
   def main(args: Array[String]): Unit = {
-    val spark = SparkSession.builder().appName("Analyze Diamond Data Spark App").getOrCreate()
+    val spark = SparkSession
+      .builder()
+      .master("local")
+      .appName("Analyze Diamond Data Spark App")
+      .getOrCreate()
 
     val diamondDF = spark.read
       .option("header", true)