Add generated models to exports for main