Fixing problem1 to return solution, not sum fn
Add solution for problem 5