Added problem_002.py, a Python solution to problem 2