updated problem 1