Update challenge1.py

punct