Update Problem2.py