Adds PySpark support