Added SparkContext.cleanup back. Not sure why it was removed before ...