problem4.py