Assn2, question 1, correct submission