Update challenge.py