problem 2
update problem 2