problem9.py added