Add solution for problem 2
Update readme with problem 2