Update Problem2.py

Minor mods made.