Update problem-6.py