Add eager loader to categories and user for units