Scala/Sort: Optimize