Add DI work