Create problem_54.py