problem 1 code