add import org.apache.spark.SparkContext._ back