Move DI config into separate class