diff --git a/Project/src/main/scala/Driver.scala b/Project/src/main/scala/Driver.scala
index 61c27f9..92a6e7d 100644
--- a/Project/src/main/scala/Driver.scala
+++ b/Project/src/main/scala/Driver.scala
@@ -58,23 +58,21 @@ object Driver {
           val end = System.nanoTime
           println("KMeans Run-Time: " + (end - start) / 10e9 + "s")
 
-          val clusterCenters = model.clusterCenters map (_.toArray)
+          val clusterCenters = model.clusterCenters map (_.toArray).collect()
           val cost = model.computeCost(parsedData)
           println("Cost: " + cost)
 
           //write cluster centers to file
-          val ccs = clusterCenters.collect()
           val writer1 = new PrintWriter(new File("uberNightClusterCenters.txt"))
-          ccs.foreach(x => writer1.write(x + "\n"))
+          clusterCenters.foreach(x => writer1.write(x + "\n"))
           writer1.close()
 
           //show cluster centers and the number of members in them
           //write to file
-          val clusterInd = model.predict(parsedData)
-          val cis = clusterInd.collect()
+          val clusterInd = model.predict(parsedData).collect()
           val clusterSizes = clusterInd.countByValue()
           val writer2 = new PrintWriter(new File("uberNightClusterSizes.txt"))
-          cis.foreach(x => writer2.write(x + "\n"))
+          clusterSizes.foreach(x => writer2.write(x + "\n"))
           writer2.close()