How To Get Other Columns When Using Spark Dataframe Groupby