Added DI registration method