Add model associations
Use Collection instead of Set for performance