added problem 4