added problem 2 and problem 3