added associations to the models