Starting to solve Problem 4