Create problem5.py