Add tiny DI