Simple solution for problem 3