Update collatz.rb