Python code to demonstrate Collatz Conjecture