Add problem1.py - Solved!