Add solution for problem number 7