fix problem 3
fix problem 3