Add Dependency Injection