Update challenge5.py