Update Problem-3.py