solved problem4!, but not good solution.