Create collatz.py