rewrite with pyspark