Added dependency injection container