Add code to prep the data