edited problem 1