Updated prob1.py to comply with PEP8.
Updated prob2.py to comply with PEP8.