Adding the DI and annotations