Added question 1 code