Added the dependency injection