Create problem 3.py