Update Problem 10.py