add problem 3
add problem 4 and 5