TODO: Problem 1 part(b)