Add solution for problem 1
Add solution for problem 2
Add solution for problem 4