TD2: Add preliminary code for question 1.