Attach models to the data source

/to @ritch