Would it be easy to make it so that you just import `pyspark.SparkContext` instead of `pyspark.context.SparkContext`? It's a minor thing but I think it would be a bit nicer. Just not sure whether Python modules will work that way.