problem7.py added