tweak problem 2 answer. make it more readable. add comment describing problem