Add solution for problem number 4