Map and FlatMap in Spark | Scala