Update challenge_2.py