added associations in the models