problem2.py