Updated DI