Preparing the data for our models