Move association definition to the model file