Changes to remove RDD's from memory