Using better DI