Create problem3.py