Made problem1.py PEP-8 compliant
Made problem2.py PEP-8 compliant