pyspark class example