Added dependency injection extension