update to problem 3 solution