Using PySpark to Read and Flatten JSON data with an enforced schema