Update challenge_3.py