added problem 3 and 4