Update challenge2.py