modified models to have relationships