Update problem6.py