Update problem003.py