Updated collatz.py to include better comments