Update Problem3.py