Setup the schema for the models