Repartition dataframe in spark scala. .