Update challenge3.py