diff --git a/src/main/scala/util/MLLRDemo.scala b/src/main/scala/util/MLLRDemo.scala
index 01e96e2..50308a6 100644
--- a/src/main/scala/util/MLLRDemo.scala
+++ b/src/main/scala/util/MLLRDemo.scala
@@ -7,7 +7,8 @@ import org.apache.spark.SparkContext
 
 object MLILRDemo {
   def main(args: Array[String]) = {
-    val mc = new MLContext(new SparkContext("local[4]", "MLILRtest"))
+    val sc = new SparkContext("local[4]", "MLIRtest")
+    val mc = new MLContext(sc)
 
     val data = mc.loadCsvFile(args(0))
     val d2 = data.map((x: MLRow) => x.drop(0).+:(if(x(0).toString == "n07760859") MLValue(1.0) else MLValue(0.0))).cache()
@@ -16,5 +17,6 @@ object MLILRDemo {
     //val model = SVMAlgorithm.train(d2)
 
     println("Time to train: " + model.trainingTime)
+    sc.stop()
   }
 }
\ No newline at end of file