Create problem_021.py