Update problem9.py