problem3.py