Update python_challenge.py