add problem 3 and 4