Update challenge_1.py