problem 2 answer edited