Add DI helper + some doc