Exercise 5: Prepare the data