Update problem8.py