Implemented and tested PySpark support