Create problem 5.py