updated challenge1.py