Update challenge6.py