scala/collatz: simpler