Added all the associations to the models. I think....