collatz-python