problem1.py