Update 3n+1 problem.py