Update challenge_4.py