ADD: solution to problem 5