Update problem-5.py