Adding collatz.py