made changes to model associations