Defining Schema for json data in Pyspark