Added problem2.py