collatz.py