Update problem005.py