pyspark practice problems hackerrank