problem 2.py