Memory Management in Pyspark