Sample code for exercise 1