Refactored DI config