Create problem-002.py