Update challenge1.py