Update problem3.py