Updated DI definition + generator