problem 2 solution
add problem 4 solution
fix problem 4 solution