Use withSparkContext also in main code