Write DataFrame into json file using PySpark in jupiter notebook