Add documentation for problem 3
Update problem 2 readme