problem6.py