Add part of problem 1