Create problem002.py