updated problem 3 and 4