Improve DI configuration