Setup the data model