Scala first n partition