Update Challenge 2.py