clean problem #3 solution
solve problem #4