Added associations between models