Added problem_3.
Wrote the rest of the code.