Add solution to 3 problem