updating challenge1.py