Update problem10.py