Create problem31.py