Add Assocciation to the models