Added associations to the model files