add problem 4 sol