Collatz Conjecture in Python