update models to add relationships