Add problem2.py, problem3.py, and problem4.py