Updates to DI